from ultralytics.models import YOLO

if __name__ == '__main__':
	model = YOLO(model='./runs/train/exp/weights/best.pt')
	model.predict(source='../datasets/test/images/', device='0', imgsz=640, project='runs/detect', name='exp', save=True)